from pydantic import BaseModel

class PaperRequest(BaseModel):
    title: str
    abstract: str

class RecommendationRequest(BaseModel):
    paper_id: int
    top_k: int = 5